Define range in python